from dotenv import load_dotenv
import os

# 加载.env文件
load_dotenv(dotenv_path="./config/.env")


class ConfigUtil:
    """
    配置工具类，用于读取环境变量配置信息
    """
    @staticmethod
    def get_test_url():
        """
        获取测试URL配置

        Returns:
            str: 测试URL地址
        """
        return os.getenv("TEST_URL")

    @staticmethod
    def get_browser():
        """
        获取浏览器配置

        Returns:
            str: 浏览器类型，默认返回"chrome"
        """
        return os.getenv("BROWSER", "chrome")  # 默认chrome

    # 数据库配置
    @staticmethod
    def get_db_config():
        """
        获取数据库配置信息

        Returns:
            dict: 包含数据库连接信息的字典，包含host、port、user、password、database字段
        """
        return {
            "host": os.getenv("DB_HOST"),
            "port": int(os.getenv("DB_PORT")),
            "user": os.getenv("DB_USER"),
            "password": os.getenv("DB_PASS"),
            "database": os.getenv("DB_NAME")
        }